"""
https://leetcode.cn/problems/ba-shu-zu-pai-cheng-zui-xiao-de-shu-lcof/description/

LCR 164. 破解闯关密码
中等
相关标签
相关企业
闯关游戏需要破解一组密码，闯关组给出的有关密码的线索是：

一个拥有密码所有元素的非负整数数组 password
密码是 password 中所有元素拼接后得到的最小的一个数
请编写一个程序返回这个密码。

 

示例 1:

输入: password = [15, 8, 7]
输出: "1578"
示例 2:

输入: password = [0, 3, 30, 34, 5, 9]
输出: "03033459"
 

提示:

0 < password.length <= 100
说明:

输出结果可能非常大，所以你需要返回一个字符串而不是整数
拼接起来的数字可能会有前导 0，最后结果不需要去掉前导 0
"""

from typing import List
import itertools


class Solution:
    def group_by_rule(self,lst, key_func):
        result = []
        # 使用itertools.groupby进行分组
        for key, group in itertools.groupby(lst, key_func):
            result.append(list(group))
        return result

    def sort_inner(self,bucket_num:int,nums:List[str])->List[str]:
        max_len=1
        for num in nums:
            max_len=max(max_len,len(num))

        def sort_val(val:str)->int:
            if len(val)<max_len:
                suffix=(str(bucket_num))*(max_len-len(val))
                return int(val+suffix)
            return int(val)
        
        return sorted(nums,key=sort_val)



    def crackPassword(self, password: List[int]) -> str:
        """
        由于一个最高位最小的数字排在最前面，必定比其他数字构建的结果小
        所以规则实际是按照最高到地位进行排序

        比较特殊的是，遇到不同位数但前缀相同的情况
        此时应该从其余元素中找到高位最小的元素，与之对比
        
        这种对比的后续计算量较大，所以不如直接拆分成若干分支，
        所有某个分支已经比当前维护的最高分支小，则废弃掉
        """

        num_strs=[str(_) for _ in password]
        num_strs=sorted(num_strs)

        grouped_list=self.group_by_rule(num_strs, lambda s: s[0])

        """
        实际上可以对一个不足最高位的数字进行排序，如在1中，'3'在比较中相当于'33'
        0=['0']
        1=['3', '30', '34']
        2=['5']
        3=['9']
        """
        res=[]
        for i,nums in enumerate(grouped_list):
            # print(f'{i}={self.sort_inner(int(nums[0]),nums)}')
            res.extend(self.sort_inner(int(nums[0]),nums))
            
        return ''.join(res)                 



if __name__=='__main__':
    password = [0, 3, 30, 34, 5, 9]
    sol=Solution()
    res=sol.crackPassword(password)
    print(res)
    # print('a'*5)
    pass